package com.small.common.util;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;
import java.util.zip.CRC32;
import java.util.zip.CheckedInputStream;

/**
 * author yefeng
 * time 2017/5/25 17:58
 */
public class FileHelper {

    private static final Logger LOGGER = LoggerFactory.getLogger(FileHelper.class);

    /**
     * 1kb
     */
    private final static int KB_1 = 1024;

    /**
     * @param file 要进行校验的文件
     * @return 获得文件的CRC32校验和
     * @throws Exception
     */
    public static String getFileCRCCode(File file) throws Exception {
        FileInputStream is = new FileInputStream(file);
        CRC32 crc32 = new CRC32();
        CheckedInputStream cis = new CheckedInputStream(is, crc32);
        byte[] buffer = new byte[KB_1];
        while (cis.read(buffer) != -1) {
        }
        is.close();
        return Long.toHexString(crc32.getValue());
    }

    /**
     * @param string 要进行校验的字串
     * @return 获得字串的CRC32校验和
     * @throws Exception
     */
    public static String getStringCRCCode(String string) throws Exception {
        ByteArrayInputStream inputStream = new ByteArrayInputStream(string.getBytes());
        CRC32 crc32 = new CRC32();
        CheckedInputStream checkedinputstream = new CheckedInputStream(inputStream, crc32);
        byte[] buffer = new byte[KB_1];
        while (checkedinputstream.read(buffer) != -1) {
        }
        inputStream.close();
        return Long.toHexString(crc32.getValue());
    }


    /**
     * @param basePath          文件路径
     * @param fullFilenameToAdd 文件名称
     * @return 连接路径和文件名称，组成最后的包含路径的文件名
     */
    public static String concat(String basePath, String fullFilenameToAdd) {
        return FilenameUtils.concat(basePath, fullFilenameToAdd);
    }

    /**
     * 获得不带文件扩展名的文件名称
     *
     * @param filename 文件完整路径
     * @return 不带扩展名的文件名称
     */
    public static String getBaseName(String filename) {
        return FilenameUtils.getBaseName(filename);
    }

    /**
     * 获得带扩展名的文件名称
     *
     * @param filename 文件完整路径
     * @return 文件名称
     */
    public static String getFileName(String filename) {
        return FilenameUtils.getName(filename);
    }

    /**
     * 获得文件的完整路径，包含最后的路径分隔条
     *
     * @param filename 文件完整路径
     * @return 目录结构
     */
    public static String getFullPath(String filename) {
        return FilenameUtils.getFullPath(filename);
    }

    /**
     * @param filename 文件完整路径
     * @return 获得文件的完整路径，不包含最后的路径分隔条
     */
    public static String getFullPathNoEndSeparator(String filename) {
        return FilenameUtils.getFullPathNoEndSeparator(filename);
    }

    /**
     * 判断文件是否有某扩展名
     *
     * @param filename  文件完整路径
     * @param extension 扩展名名称
     * @return 若是，返回true，否则返回false
     */
    public static boolean isExtension(String filename, String extension) {
        return FilenameUtils.isExtension(filename, extension);
    }


    /**
     * 判断文件的扩展名是否是扩展名数组中的一个
     *
     * @param filename   文件完整路径
     * @param extensions 扩展名名称
     * @return 若是，返回true，否则返回false
     */
    public static boolean isExtension(String filename, String[] extensions) {
        return FilenameUtils.isExtension(filename, extensions);
    }

    /**
     * @param filename 文件完整路径
     * @return 规范化路径，合并其中的多个分隔符为一个,并转化为本地系统路径格式
     */
    public static String normalize(String filename) {
        return FilenameUtils.normalize(filename);
    }

    /**
     * @param filename 文件完整路径
     * @return 规范化路径，合并其中的多个分隔符为一个,并转化为本地系统路径格式,若是路径，则不带最后的路径分隔符
     */
    public static String normalizeNoEndSeparator(String filename) {
        return FilenameUtils.normalizeNoEndSeparator(filename);
    }

    /**
     * 把文件路径中的分隔符转换为unix的格式，也就是"/"
     *
     * @param path 文件完整路径
     * @return 转换后的路径
     */
    public static String separatorsToUnix(String path) {
        return FilenameUtils.separatorsToUnix(path);
    }

    /**
     * 把文件路径中的分隔符转换为window的格式，也就是"\"
     *
     * @param path 文件完整路径
     * @return 转换后的路径
     */
    public static String separatorsToWindows(String path) {
        return FilenameUtils.separatorsToWindows(path);
    }

    /**
     * 把文件路径中的分隔符转换当前系统的分隔符
     *
     * @param path 文件完整路径
     * @return 转换后的路径
     */
    public static String separatorsToSystem(String path) {
        return FilenameUtils.separatorsToSystem(path);
    }


    /**
     * 提取文件的扩展名
     *
     * @param filename 文件名称
     * @return 文件扩展名，若没有扩展名，则返回空字符串
     */
    public static String getExtension(String filename) {
        return FilenameUtils.getExtension(filename);
    }

    /**
     * 移出文件的扩展名
     *
     * @param filename 文件名称
     * @return 若文件存在扩展名，则移出扩展名，然后返回移出后的值
     */
    public static String removeExtension(String filename) {
        return FilenameUtils.removeExtension(filename);
    }


    /**
     * 清除一个目录的内容，但不删除此目录
     *
     * @param directory 需要清除的目录
     * @return true:清除成功   false:清除失败
     */
    public static boolean cleanDirectory(File directory) {
        try {
            FileUtils.cleanDirectory(directory);
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }


    /**
     * 拷贝一个目录的内容到另外一个目录中
     *
     * @param srcDir  源目录
     * @param destDir 目的目录
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyDirectory(File srcDir, File destDir) {
        return copyDirectory(srcDir, destDir, true);
    }

    /**
     * 拷贝一个目录的内容到另外一个目录中
     *
     * @param srcDir  源目录
     * @param destDir 目的目录
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyDirectory(String srcDir, String destDir) {
        return copyDirectory(new File(srcDir), new File(destDir));
    }

    /**
     * 拷贝一个目录的内容到另外一个目录中
     *
     * @param srcDir           源目录
     * @param destDir          目的目录
     * @param preserveFileDate 是否保持文件日期
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyDirectory(File srcDir, File destDir, boolean preserveFileDate) {
        try {
            FileUtils.copyDirectory(srcDir, destDir, preserveFileDate);
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 拷贝源目录的内容到目的目录中(注：是拷贝到目的目录的里面)
     *
     * @param srcDir  源目录
     * @param destDir 目的目录
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyDirectoryToDirectory(File srcDir, File destDir) {
        try {
            FileUtils.copyDirectoryToDirectory(srcDir, destDir);
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 拷贝源目录的内容到目的目录中(注：是拷贝到目的目录的里面)
     *
     * @param srcDir  源目录
     * @param destDir 目的目录
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyDirectoryToDirectory(String srcDir, String destDir) {
        return copyDirectoryToDirectory(new File(srcDir), new File(destDir));
    }

    /**
     * 拷贝文件
     *
     * @param srcFile  源文件
     * @param destFile 目的文件
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyFile(File srcFile, File destFile) {
        return copyFile(srcFile, destFile, true);
    }

    /**
     * 拷贝文件
     *
     * @param srcFile  源文件路径
     * @param destFile 目的文件路径
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyFile(String srcFile, String destFile) {
        return copyFile(new File(srcFile), new File(destFile));
    }

    /**
     * 拷贝文件
     *
     * @param srcFile          源文件
     * @param destFile         目的文件
     * @param preserveFileDate 是否保留文件日期
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyFile(File srcFile, File destFile, boolean preserveFileDate) {
        try {
            FileUtils.copyFile(srcFile, destFile, preserveFileDate);
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 拷贝文件到某目录中
     *
     * @param srcFile 源文件
     * @param destDir 目的目录
     * @return true:拷贝成功   false:拷贝失败
     */
    public static boolean copyFileToDirectory(File srcFile, File destDir) {
        try {
            FileUtils.copyFileToDirectory(srcFile, destDir);
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 拷贝文件到某目录中
     *
     * @param srcFile 源文件
     * @param destDir 目的目录
     * @return true:拷贝成功    false:拷贝失败
     */
    public static boolean copyFileToDirectory(String srcFile, String destDir) {
        return copyFileToDirectory(new File(srcFile), new File(destDir));
    }

    /**
     * 删除一个目录和该目录下的所有内容
     *
     * @param directory 需要删除的目录
     * @return true:删除成功    false:删除失败
     */
    public static boolean deleteDirectory(String directory) {
        try {
            FileUtils.deleteDirectory(new File(directory));
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 删除文件
     *
     * @param path 需要删除的文件路径
     */
    public static void deleteFile(String path) {
        try {
            File file = new File(path);
            if (file.exists()) {
                FileUtils.forceDelete(file);
            }
        } catch (IOException ex) {
            LOGGER.error("删除文件异常", ex);
        }
    }

    /**
     * 删除文件或目录
     *
     * @param file 需要删除的文件路径
     * @return true:删除成功    false:删除失败
     */
    public static boolean delete(String file) {
        try {
            File f = new File(file);
            if (f.isFile()) {
                FileUtils.forceDelete(f);
            } else {
                FileUtils.deleteDirectory(f);
            }
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * @param directory 目录
     * @return 递归创建目录
     */
    public static boolean createDirectory(String directory) {
        try {
            FileUtils.forceMkdir(new File(directory));
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 读入文件到字节数组中
     *
     * @param file 需要读取的文件路径
     * @return 读取的字节数组，若读入失败，则返回null
     */
    public static byte[] readFileToByteArray(String file) {
        try {
            return FileUtils.readFileToByteArray(new File(file));
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return null;
    }

    /**
     * 读入文件到字串中
     *
     * @param file 需要读取的文件路径
     * @return 读取的文件内容，若读入失败，则返回空字串
     */
    public static String readFileToString(String file) {
        try {
            return FileUtils.readFileToString(new File(file), Encoding.UTF8);
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return "";
    }

    /**
     * 读入文本文件到一个按行分开的List中
     *
     * @param file 需要读取的文件路径
     * @return 按行内容分开的List
     */
    public static List readLines(String file) {
        try {
            return FileUtils.readLines(new File(file), Encoding.UTF8);
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return null;
    }

    /**
     * 递归求一个目录的容量大小
     *
     * @param directory 需要计算容量的目录路径
     * @return 容量的大小(字节数)
     */
    public static long sizeOfDirectory(String directory) {
        return FileUtils.sizeOfDirectory(new File(directory));
    }

    /**
     * 写字节数组到文件中，若文件不存在，则建立新文件
     *
     * @param file 需要写的文件的路径
     * @param data 需要写入的字节数据
     * @return true:写入成功   false:写入失败
     */
    public static boolean writeToFile(String file, byte[] data) {
        try {
            FileUtils.writeByteArrayToFile(new File(file), data);
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 写字串到文件中，若文件不存在，则建立新文件
     *
     * @param file 需要写的文件的路径
     * @param data 需要写入的字串
     * @return true:写入成功   false:写入失败
     */
    public static boolean writeToFile(String file, String data) {
        return writeToFile(file, data, Encoding.UTF8);
    }

    /**
     * 写字串到文件中，若文件不存在，则建立新文件
     *
     * @param file     需要写的文件的路径
     * @param data     需要写入的字串
     * @param encoding 文件编码，默认为UTF-8
     * @return true:写入成功   false:写入失败
     */
    public static boolean writeToFile(String file, String data, String encoding) {
        try {
            FileUtils.writeStringToFile(new File(file), data, encoding);
            return true;
        } catch (IOException ex) {
            LOGGER.error("", ex);
        }
        return false;
    }

    /**
     * 把文件写入到response的输出流里，可用来做文件下载
     *
     * @param response HttpServletResponse
     * @param filePath 文件路径
     */
    public static void writeToResponse(HttpServletResponse response, String filePath) {
        writeToResponse(response, filePath, 1024);
    }


    /**
     * 把文件写入到response的输出流里，可用来做文件下载
     *
     * @param response   HttpServletResponse
     * @param filePath   文件路径
     * @param byteLength 缓存空间byte[]长度
     */
    public static void writeToResponse(HttpServletResponse response, String filePath, int byteLength) {
        try {
            byte[] buffer = new byte[byteLength]; //  缓冲区
            BufferedOutputStream output = null;
            BufferedInputStream input = null;
            try {
                output = new BufferedOutputStream(response.getOutputStream());
                input = new BufferedInputStream(new FileInputStream(new File(filePath)));
                int n = (-1);
                while ((n = input.read(buffer, 0, 4096)) > -1) {
                    output.write(buffer, 0, n);
                }
                response.flushBuffer();
            } catch (Exception e) {
                LOGGER.error(e.getMessage(), e);
            } finally {
                if (input != null) {
                    input.close();
                }
                if (output != null) {
                    output.close();
                }
            }
        } catch (Exception e) {
            LOGGER.error(e.getMessage(), e);
        }
    }

    /**
     * @param filePathName 文件路径全名
     * @return 建立由filePathName指定的文件，若文件路径中的目录不存在，则先建立目录
     */
    public static boolean createNewFile(String filePathName) {
        String filePath = FileHelper.getFullPath(filePathName);
        //若目录不存在，则建立目录
        if (!FileHelper.exists(filePath)) {
            if (!createDirectory(filePath)) {
                return false;
            }
        }
        try {
            File file = new File(filePathName);
            return file.createNewFile();
        } catch (IOException ex) {
            LOGGER.error("", ex);
            return false;
        }
    }


    /**
     * 判断文件和目录是否已存在
     *
     * @param filePath 文件和目录完整路径
     * @return true:存在  false：不存在
     */
    public static boolean exists(String filePath) {
        File file = new File(filePath);
        return file.exists();
    }

    /**
     * 判断特定的路径是否为文件
     *
     * @param filePath 文件完整的路径
     * @return 若是文件，则返回true，否则返回false
     */
    public static boolean isFile(String filePath) {
        File file = new File(filePath);
        return file.isFile();
    }

    /**
     * 判断特定的路径是否为目录
     *
     * @param filePath 文件完整的路径
     * @return 若是目录，则返回true，否则返回false
     */
    public static boolean isDirectory(String filePath) {
        File file = new File(filePath);
        return file.isDirectory();
    }

    /**
     * @param srcFile  源文件路径名称
     * @param destFile 目的文件路径名称
     * @return 更改文件的名称，若不在同一个目录下,则系统会移动文件
     */
    public static boolean renameTo(String srcFile, String destFile) {
        File file = new File(srcFile);
        return file.renameTo(new File(destFile));
    }

}
